ERROR: Pytorch - BrokenPipeError:[Errno 32] Broken pipe 해결방법
케이 (3)2021년 11월 06일
Tags #python#error#Torch

image

colab에서는 분명 잘 되었던 건데 윈도우에서 자체 세팅한 jupyter로 옮기고 실행하다가 해당 에러를 만나셨나요?

윈도우에서는 torch.utils.data.DataLoader()의 num_workers 값이 0이 아니면 해당 에러가 나타날 수 있습니다.

아래처럼 값을 0으로 변경해주면 정상적으로 동작하게 됩니다.

torch.utils.data.DataLoader(train_data, batch_size=batch_size, num_workers=0)

0개의 댓글